anaconda安装tensorflow

您所在的位置:网站首页 检查tensorflow gpu安装 anaconda安装tensorflow

anaconda安装tensorflow

#anaconda安装tensorflow| 来源: 网络整理| 查看: 265

保姆级手把手教你安装TensorFlow-GPU,避免坑 安装TensorFlow-GPU(bb几句,大佬勿喷)检查自己的电脑是否能安装GPU版本的==好了,这里说一下,一定要看,一定要看,一定要看一定要看!!!!!四遍了哦!==anaconda开始安装tf-gpu完成之后,下载安装CUDA和CUDNN==其实最烦的是下载这一步== 我这里已经说了方法检测cuda的安装检测tf好了,到这里就结束

安装TensorFlow-GPU(bb几句,大佬勿喷)

因为最近要弄点机器学习(人脸表情项目)的东西,所以得安装TensorFlow,我本来是A卡的电脑,还特意换了台N卡电脑(偷偷bb:主要也是我的那台当时实体店被坑了老惨,现在终于换了,放家里看电视或者跑数据吧)。能搜到的tensorflow(后面简写tf)的安装教程很多都是旧的,还是有挺多坑了,我也终于在一篇又一篇的文章中成功安装好了。

检查自己的电脑是否能安装GPU版本的

其实,能够用到这些,基本也不会太小白了。不过我还是说一下吧。首先你的电脑有独立显卡,而且是N卡即NVIDIA的显卡。(A卡AMD)

查显卡 ,电脑右键属性,设备管理器,显示适配器;我的是GTX1650Ti在这里插入图片描述 这里说一下,NVIDIA的官网给出支持显卡驱动的表格没有1650Ti的,但我我能使用,搞不懂,可能是高于某些型号的都行吧。这里可以查一下你的显卡是否支持。https://developer.nvidia.com/cuda-gpus比较懒的我就在这里贴出来了 在这里插入图片描述 我的是GeForce的 在这里插入图片描述 在这里插入图片描述 在这里插入图片描述 在这里插入图片描述 好了,这里说一下,一定要看,一定要看,一定要看一定要看!!!!!四遍了哦!

对应版本!! 对应版本!! 对应版本!!

注意安装的tf版本及其对应的cuda和cudnn版本,官网有。https://tensorflow.google.cn/install/source_windows#cpu 如果没有最新版的换成英文就可以了。在这里插入图片描述 我安装的是tf2.4.1+cuda11.0+cudnn8.0(图片虽然没有2.4.1,但我安装是可以用的)

anaconda开始安装tf-gpu

打开anaconda的prompt 这里我参考了这位大哥的博客,镜像是清华的

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu

默认安装最新版本,我当时是2.4.1。如果要安装其他版本,参考以下

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu=2.3

完成之后,下载安装CUDA和CUDNN其实最烦的是下载这一步 我这里已经说了方法

CUDA下载,打开有点慢,下载也是https://developer.nvidia.com/cuda-toolkit-archive,不行的话右键复制链接用某雷下载(安装官网某雷时注意捆绑软件) CUDNN下载这个打开也慢https://developer.nvidia.com/rdp/cudnn-archive而且还得注册账户,如果网不行就改一下网络适配WIFI4协议的DNS,登录后填个问卷就可以了。下载可以类似右键复制链接,用某雷。(超级加速6块)(偷偷bb几句,某雷似乎挺强大的,可以找白嫖某雷会员的资源)手动滑稽

安装cuda,路径记一下配环境的bin,这个我就不写了,给这位大哥的博客

检测cuda的安装

依旧是prompt

nvcc -V

在这里插入图片描述 显示这样就是成功

检测tf

键入python

import tensorflow as tf tf.test.gpu_device_name()

是否显示显卡型号 或者

import tensorflow as tf print(tf.test.is_gpu_available())

是否true

好了,到这里就结束


【本文地址】


今日新闻


推荐新闻


CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3